package org.jtzc.langchang4jdemo;

import dev.langchain4j.model.chat.ChatLanguageModel;
import dev.langchain4j.model.openai.OpenAiChatModel;

/**
 * @author wu chuang
 * @description
 */
public class HelloAI {

    public static void main(String[] args) {

//        ChatLanguageModel model = OpenAiChatModel.withApiKey("demo");
//
//        String answer = model.generate("你好，你是谁？");
//
//        System.out.println(answer);

        //初始化模型
        OpenAiChatModel model = OpenAiChatModel.builder()
                //LangChain4j提供的代理服务器，该代理服务器会将演示密钥替换成真实密钥， 再将请求转发给OpenAI API
                .baseUrl("http://langchain4j.dev/demo/openai/v1") //设置模型api地址（如果apiKey="demo"，则可省略baseUrl的配置）
                .apiKey("demo") //设置模型apiKey
                .modelName("gpt-4o-mini") //设置模型名称
                .build();

        //向模型提问
//        String answer = model.generate("请问今天是星期几?什么时候放假?");
        String answer = model.chat("请问今天是星期几?什么时候放假?");
        //输出结果
        System.out.println(answer);

    }

}
